# -*- coding: utf-8 -*-
from hotel import *

hotel = AnimalHotel()

print("欢迎来到小草神酒店管理系统")
print("知识")
print("与你分享！")

def main():
    while True:
        print("欢迎使用动物旅馆管理系统！请选择您要进行的操作：")
        print("1. 添加房间")
        print("2. 删除房间")
        print("3. 入住动物")
        print("4. 退房")
        print("5. 查看所有房间信息")
        print("6. 保存房间信息")
        print("7. 读取文件中的房间信息")
        print("8. 退出系统")
        choice = input()
        if choice == "1":
            hotel.add_room()
        elif choice == "2":
            hotel.remove_room()
        elif choice == "3":
            hotel.register_proper_room()
        elif choice == "4":
            hotel.check_out_of_room()
        elif choice == "5":
            hotel.view_all_rooms()
        elif choice == "6":
            hotel.save_all_room()
        elif choice == "7":
            hotel.get_all_rooms_from_csv()
        elif choice == "8":
            print("欢迎再次使用！")
            print("别忘记给小草神一个三连哦")
            break
        else:
            print("输入有误！请重新选择。")

if __name__ == "__main__":
    main()
